QUASAR: Lowering the Loss Floor of Quantization-Aware Training with Loss-Aware Reconstruction
QUASAR:通过损失感知重建降低量化感知训练的损失下界
🧭 量化感知训练 · 损失感知重建 ★ 深度解读
Cornell University · Together AI
将损失感知重建引入 QAT 训练循环,用 Adam 二阶矩做 saliency 进行尺度搜索和加权最小二乘拟合,压低低比特模型 loss floor
📌 工作参考价值(4 点)
- 低比特部署 LLM 时,若线上沿用 PTQ 掉点明显,可切换到 QUASAR 式 QAT:保持 weight-only 量化,不增加推理 overhead,训练开销仅 +1.4%。对需要长期稳定低延迟的推荐/Agent 推理服务有直接落地价值。
- 可直接复用两个 trick:用 AdamW 的 second moment \(v_t\) 作为权重的 saliency 估计,无需额外 Hessian 计算/存储;每步在 [0.3,1.0] 范围内搜索裁剪因子,并对反量化 scale/zero-point 做 saliency 加权最小二乘闭式求解,插入现有训练循环成本极低。
- 在 2-bit 这类极端低精度下,QAT 对数学推理和长链 Agent 任务质量影响显著;QUASAR 直接低比特 SFT 平均数学基准比最强 baseline 高 10.9 点,且是唯一在 HMMT'25 上非零的方法,说明保住长生成能力很关键。
- 该方法兼容 NVFP4 等生产格式,可把同样思路迁移到 MXFP4 或自定义 block-scaled 格式;迁移时只需调整候选 scale 与网格约束,核心目标不变。